package com.example.impandexp.controller;

import com.example.impandexp.pojo.Test;
import com.example.impandexp.utils.ExcelUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.List;

/**
 * 测试导入导出接口
 * @author xieyingde
 */
@RestController
@RequestMapping("test")
public class TestController {

    /**
     * excle导入
     * @param file 文件
     * @return
     */
    @PostMapping("/impExcel")
    public String impUser(@RequestParam("file") MultipartFile file){
        // 项目文件里面的test.xlsx 是模板文件
        System.out.println(file);
        List<Test> testList = ExcelUtils.importData(file, 1, Test.class);
        System.out.println("获取导入的test数据：" + testList);
        return "success";
    }

    /**
     * 导出excel
     * @param response
     */
    @GetMapping("expExcel")
    public void expExcel(HttpServletResponse response) {
        List<Test> testList = new ArrayList<>();
        // 这里封装数据 没有用数据库

        // 循环加入100条数据
        for (int i = 0; i < 100; i++) {
            Test test = new Test();
            test.setId(i+1);
            test.setName("小德" + i);
            test.setMobile("1889834" + i);
            test.setAddress("广东技术师范大学天河学院" + i);
            testList.add(i,test);
        }
        // 导出表格数据
        if (testList != null && testList.size() > 0) {
            ExcelUtils.exportExcel(testList,"导出","测试导出",Test.class,"test.xlsx",response);
        }
    }
}
